package com.rhythm.commons.security;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.security.spec.AlgorithmParameterSpec;
import java.security.spec.KeySpec;
import javax.crypto.Cipher;
import javax.crypto.CipherInputStream;
import javax.crypto.CipherOutputStream;
import javax.crypto.IllegalBlockSizeException;
import javax.crypto.SecretKey;
import javax.crypto.SecretKeyFactory;
import javax.crypto.spec.PBEKeySpec;
import javax.crypto.spec.PBEParameterSpec;

/**
 *
 * @author Michael J. Lee @ Synergy Energy Holdings, LLC
 */
public class DESCrypter
{

    private Cipher ecipher;

    private Cipher dcipher;
    // 8-byte Salt
    byte[] salt =
    {
        (byte) 0xA9, (byte) 0x9B, (byte) 0xC8, (byte) 0x32,
        (byte) 0x56, (byte) 0x35, (byte) 0xE3, (byte) 0x03
    };
    // Iteration count
    int iterationCount = 19;
    
    public DESCrypter(String passPhrase) {
        try
        {
            // Create the key
            KeySpec keySpec = new PBEKeySpec(passPhrase.toCharArray(), salt, iterationCount);
            SecretKey key = SecretKeyFactory.getInstance("PBEWithMD5AndDES").generateSecret(keySpec);
            
            ecipher = Cipher.getInstance(key.getAlgorithm());
            dcipher = Cipher.getInstance(key.getAlgorithm());

            // Prepare the parameter to the ciphers
            AlgorithmParameterSpec paramSpec = new PBEParameterSpec(salt, iterationCount);

            // Create the ciphers
            ecipher.init(Cipher.ENCRYPT_MODE, key, paramSpec);
            dcipher.init(Cipher.DECRYPT_MODE, key, paramSpec);
        } catch (java.security.InvalidAlgorithmParameterException e)
        {
        } catch (java.security.spec.InvalidKeySpecException e)
        {
        } catch (javax.crypto.NoSuchPaddingException e)
        {
        } catch (java.security.NoSuchAlgorithmException e)
        {
        } catch (java.security.InvalidKeyException e)
        {
        }
    }
    byte[] buf = new byte[1024];

    public void encrypt(InputStream in, OutputStream out) throws IOException {

        try
        {
            // Bytes written to out will be encrypted
            out = new CipherOutputStream(out, ecipher);

            // Read in the cleartext bytes and write to out to encrypt
            int numRead = 0;
            while ((numRead = in.read(buf)) >= 0)
            {
                out.write(buf, 0, numRead);
            }

        } finally
        {
            if (out != null)
                out.close();

            if (in != null)
                in.close();
        }
    }

    public String encrypt(String str) {
        try
        {
            // Encode the string into bytes using utf-8
            byte[] utf8 = str.getBytes("UTF8");

            // Encrypt
            byte[] enc = ecipher.doFinal(utf8);

            // Encode bytes to base64 to get a string

            return String.valueOf(Base64Coder.encode(enc));
        //return new sun.misc.BASE64Encoder().encode(enc);
        } catch (javax.crypto.BadPaddingException e)
        {
        } catch (IllegalBlockSizeException e)
        {
        } catch (UnsupportedEncodingException e)
        {
        } catch (java.io.IOException e)
        {
        }
        return null;
    }

    public String decrypt(String str) {
        try
        {
            byte[] dec = Base64Coder.decode(str);
            byte[] utf8 = dcipher.doFinal(dec);
            return new String(utf8,"UTF8");
            
        } catch (javax.crypto.BadPaddingException e)
        {
        } catch (IllegalBlockSizeException e)
        {
        } catch (UnsupportedEncodingException e)
        {
        } catch (java.io.IOException e)
        {
        }
        return null;
    }

    public void decrypt(InputStream in, OutputStream out) throws IOException {
        try
        {
            // Bytes read from in will be decrypted
            in = new CipherInputStream(in, dcipher);

            // Read in the decrypted bytes and write the cleartext to out
            int numRead = 0;
            while ((numRead = in.read(buf)) >= 0)
            {
                out.write(buf,
                     0, numRead);
            }
        } finally
        {
            if (out != null)
                out.close();

            if (in != null)
                in.close();
        }
    }
}
